/*
题目描述
有一个二维数组(n*n),写程序实现从右上角到左下角沿主对角线方向打印。
给定一个二位数组arr及题目中的参数n，请返回结果数组。
测试样例：
[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]],4
返回：[4,3,8,2,7,12,1,6,11,16,5,10,15,9,14,13]
*/
class Printer {
public:
    vector<int> arrayPrint(vector<vector<int> > arr, int n) {
        // write code here
        vector <int>v;
        for (int i = n - 1;i >= 0;i --)
        {
            int x = 0;
            int y = i;
            while (x < n && y < n)
                v.push_back(arr[x ++][y ++]);
        }
        for (int i = 1;i < n;i ++)
        {
            int x = 1;
            int y = 0;
            while (x < n && y < n)
                v.push_back(arr[x ++][y ++]);
        }
        return v;
    }
};